#!/bin/bash
# TCD-VoIP 音频质量评估系统运行脚本

echo "========================================"
echo "TCD-VoIP 音频质量评估系统"
echo "========================================"

# 检查audio环境是否存在
if ! conda info --envs | grep -q "audio"; then
    echo "错误: 找不到conda环境 'audio'"
    echo "请先创建并安装依赖: conda create -n audio python=3.11"
    exit 1
fi

# 检查编译的工具是否存在
echo "1. 检查音频质量评估工具..."
if [ ! -f "./tools/build/bin/pesq" ]; then
    echo "警告: PESQ工具未找到"
    echo "请先编译工具: cd tools/build && cmake .. && make"
fi

if [ ! -f "./tools/build/bin/visqol" ]; then
    echo "警告: ViSQOL工具未找到"
fi

if [ ! -f "./tools/build/bin/p563" ]; then
    echo "警告: P.563工具未找到"
fi

# 检查数据集
echo "2. 检查数据集..."
if [ ! -d "./dataset/Test Set" ]; then
    echo "警告: 数据集目录未找到 './dataset/Test Set'"
    echo "请确保TCD-VoIP数据集已正确放置"
fi

# 运行评估
echo "3. 开始音频质量评估..."
echo "激活conda环境 'audio' 并运行评估脚本, 样本限制为10个(如需更改自行调整)..."

conda run -n audio python run_evaluation.py --sample-limit 10

echo ""
echo "4. 如果评估完成，运行分析脚本..."
if [ -f "evaluation_results.csv" ]; then
    conda run -n audio python analyze_results.py
    echo ""
    echo "分析完成！结果保存在 'analysis_results/' 目录中"
else
    echo "评估结果文件未生成，请检查评估过程中的错误"
fi

echo "========================================"
echo "处理完成"
echo "========================================"
